# 图像通道组合

# encoding:utf-8
import cv2
import numpy as np
import matplotlib.pyplot as plt

img = cv2.imread('src/lena.png')
img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 1、灰度图像
plt.subplot(221), plt.imshow(img), plt.title('img'), plt.axis('off')  # 坐标轴关闭
plt.subplot(222), plt.imshow(img, cmap=plt.cm.gray), plt.title('img_cmap'), plt.axis('off')

plt.subplot(223), plt.imshow(img_gray), plt.title('img_gray'), plt.axis('off')
plt.subplot(224), plt.imshow(img_gray, cmap=plt.cm.gray), plt.title('img_gray_cmap'), plt.axis('off')  # 正确用法

plt.show()

# 2、彩色图像
b, g, r = cv2.split(img)  # 通道分割
img_RGB = cv2.merge([r, g, b])  # 通道组合

plt.subplot(121), plt.imshow(img), plt.title('img_BGR'), plt.axis('off')  # 坐标轴关闭
plt.subplot(122), plt.imshow(img_RGB), plt.title('img_RGB'), plt.axis('off')

plt.show()